"""
画两个曲线图上下排列
"""

from matplotlib import pyplot as plt

if __name__ == '__main__':
    batches = []
    num_batches = 20
    for i in range(num_batches):
        batches.append(i + 1)
    # plot 1:
    plt.subplot(2, 1, 1)  # 2代表两行，1代表1列，1代表这是第1个图。排列顺序为从左到右，再从上到下

    """batches,和train_loss_list长度需要相同"""
    plt.plot(batches, train_loss_list, 'r')  # train_loss_list是列表
    plt.plot(batches, train_acc_list, 'b')
    plt.plot(batches, train_spe_list, 'g')
    plt.plot(batches, train_sen_list, 'c')
    plt.title('Train Curve ')
    plt.xlabel("Batch")
    plt.ylabel("Target")
    plt.legend(["TrainLoss", "TrainAcc", "TrainSpe", "TrainSen"])
    # plt.figure()
    # plt.show()

    # plot 2:
    plt.subplot(2, 1, 2)  # 2代表两行，1代表1列，2代表这是第1个图。排列顺序为从左到右，再从上到下

    plt.plot(batches, val_loss_list, 'r')
    plt.plot(batches, val_acc_list, 'b')
    plt.plot(batches, val_spe_list, 'g')
    plt.plot(batches, val_sen_list, 'c')
    plt.title('Val Curve ')
    plt.xlabel("Batch")
    plt.ylabel("Target")
    plt.legend(["ValLoss", "ValAcc", "TrainSpe", "ValSen"])
    plt.figure()
    plt.show()

    """
    可用颜色
    r——red
    b——blue
    c——cyan
    g——green
    k——black
    w——white
    y——yellow
    m——magenta
    """
